package ir;

import java.util.HashSet;
import java.util.Set;


public class SoapRootNode {

  private Set<DataNode> dataNodes;
  private Set<StubNode> stubNodes;
  
  public SoapRootNode() {
    dataNodes = new HashSet<DataNode>();
    stubNodes = new HashSet<StubNode>();
  }
  
  public void addDataNode(DataNode d) {
    dataNodes.add(d);
  }

  public void addStubNode(StubNode stubNode) {
    stubNodes.add(stubNode); 
  }
  
  public Set<ClassGenerator> getChildren() {
    HashSet<ClassGenerator> children = new HashSet<ClassGenerator>();
    
    children.addAll(dataNodes);
    children.addAll(stubNodes);
    
    return children;
  }
  
  public Set<DataNode> getDataNodes() {
    return dataNodes;
  }
  
  public Set<StubNode> getStubNodes() {
    return stubNodes;
  }

}
